import torch
from torch import nn

criterion = nn.BCEWithLogitsLoss()
x = torch.tensor([0.1,0.1])
y = torch.tensor([0,0], dtype=torch.long)
loss = criterion(x,y)
print(loss)
